from ultralytics import YOLO

# 加载 YOLO 模型
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  # 从 YAML 构建并转移权重

if __name__ == '__main__':
    # 训练模型
    results = model.train(data='coco128.yaml', epochs=10, device='cpu')
    # 验证模型
    metrics = model.val()
